package com.ktjy.base;

import lombok.Data;
import lombok.ToString;

import java.io.Serial;
import java.io.Serializable;
import java.util.List;

/**
 * R 通用数据响应对象
 *
 * @author LILIBO
 * @since 2021-06-24
 */
@Data
@ToString
public class R<T> implements Serializable {

    @Serial
    private static final long serialVersionUID = 1L;

    /**
     * 成功状态码
     */
    public static final int CODE_SUCCESS = 0;
    /**
     * 失败状态码
     */
    public static final int CODE_FAILURE = 1;

    /**
     * 成功消息
     */
    public static final String MSG_SUCCESS = "success";
    /**
     * 失败消息
     */
    public static final String MSG_FAILURE = "failure";

    /**
     * 状态码 0: 成功; 1: 失败
     */
    private int code;

    /**
     * 消息
     */
    private String msg;

    /**
     * 数据量
     */
    private long count;

    /**
     * 数据
     */
    private T data;

    /**
     * 状态码
     *
     * @param code 状态码
     * @return 返回对象
     */
    public R<T> code(int code) {
        this.code = code;
        return this;
    }

    /**
     * 消息
     *
     * @param msg 消息
     * @return 返回对象
     */
    public R<T> msg(String msg) {
        this.msg = msg;
        return this;
    }

    /**
     * 数据量
     *
     * @param count 数据量
     * @return 返回对象
     */
    public R<T> count(long count) {
        this.count = count;
        return this;
    }

    /**
     * 数据
     *
     * @param data 数据
     * @return 返回对象
     */
    public R<T> data(T data) {
        this.data = data;
        return this;
    }

    /**
     * 返回成功
     *
     * @return
     */
    public static R success() {
        return success(CODE_SUCCESS, MSG_SUCCESS);
    }

    /**
     * 返回成功
     *
     * @param data 数据
     * @param <T>
     * @return
     */
    public static <T> R<T> success(T data) {
        return success(MSG_SUCCESS, data);
    }

    /**
     * 返回成功
     *
     * @param msg 消息
     * @param msg 数据
     * @param <T>
     * @return
     */
    public static <T> R<T> success(int code, String msg) {
        return result(code, msg, 0L, null);
    }

    /**
     * 返回成功
     *
     * @param msg 消息
     * @param data 数据
     * @param <T>
     * @return
     */
    public static <T> R<T> success(String msg, T data) {
        long count = data instanceof List<?> ? ((List<?>) data).size() : 0L;
        return result(CODE_SUCCESS, msg, count, data);
    }

    /**
     * 返回成功
     *
     * @param count 数量
     * @param data 数据
     * @param <T>
     * @return
     */
    public static <T> R<T> success(long count, T data) {
        return result(CODE_SUCCESS, MSG_SUCCESS, count, data);
    }

    /**
     * 返回失败
     *
     * @return
     */
    public static R failure() {
        return failure(CODE_FAILURE, MSG_FAILURE);
    }

    /**
     * 返回失败
     *
     * @param data 数据
     * @param <T>
     * @return
     */
    public static <T> R<T> failure(T data) {
        return failure(MSG_FAILURE, data);
    }

    /**
     * 返回失败
     *
     * @param code 状态码
     * @param msg 消息
     * @param <T>
     * @return
     */
    public static <T> R<T> failure(int code, String msg) {
        return result(code, msg, 0, null);
    }

    /**
     * 返回失败
     *
     * @param msg 消息
     * @param data 数据
     * @param <T>
     * @return
     */
    public static <T> R<T> failure(String msg, T data) {
        int count = data instanceof List<?> ? ((List<?>) data).size() : 0;
        return result(CODE_FAILURE, msg, count, data);
    }

    /**
     * 返回失败
     *
     * @param count 数量
     * @param data 数据
     * @param <T>
     * @return
     */
    public static <T> R<T> failure(long count, T data) {
        return result(CODE_FAILURE, MSG_FAILURE, count, data);
    }

    /**
     * 返回结果
     *
     * @param code 状态码
     * @param msg 消息
     * @param count 数据量
     * @param data 数据
     * @param <T>
     * @return
     */
    public static <T> R<T> result(int code, String msg, long count, T data) {
        return new R<T>().code(code).msg(msg).count(count).data(data);
    }

}
